package huawei;

/**
 * 给定一个非空字符串S，其被N个‘-’分隔成N+1的子串，给定正整数K，要求除第一个子串外，其余的子串每K个字符组成新的子串，并用‘-’分隔。对于新组成的每一个子串，如果它含有的小写字母比大写字母多，则将这个子串的所有大写字母转换为小写字母；反之，如果它含有的大写字母比小写字母多，则将这个子串的所有小写字母转换为大写字母；大小写字母的数量相等时，不做转换。
 * 输入描述:
 * 输入为两行，第一行为参数K，第二行为字符串S。
 * 输出描述:
 * 输出转换后的字符串。
 * 示例1
 * 输入
 * 3
 * 12abc-abCABc-4aB@
 * 输出
 * 12abc-abc-ABC-4aB-@
 * 说明
 * 子串为12abc、abCABc、4aB@，第一个子串保留，
 * 后面的子串每3个字符一组为abC、ABc、4aB、@，abC中小写字母较多，转换为abc，
 * ABc中大写字母较多，转换为ABC，4aB中大小写字母都为1个，不做转换，@中没有字母，连起来即12abc-abc-ABC-4aB-@
 *
 * 作者：yaozi
 * 链接：https://leetcode.cn/circle/discuss/niKSMZ/
 * 来源：力扣（LeetCode）
 * 著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。
 */
public class Question1 {

    public String splitString(String s, int k) {
        if (s == null || s.isEmpty()) {
            return "";
        }

        StringBuilder sb = new StringBuilder();

        // 查找第一个分割符号
        int fi = -1;
        for (int i = 0; i < s.length(); i++) {
            if (s.charAt(i) == '-') {
                fi = i;
                break;
            }
        }

        // 没有一个分割符号
        if (fi == -1) {
            return s;
        }

        sb.append(s, 0, fi + 1);

        int countDx = 0;
        int countXx = 0;
        int countSum = 0;

        StringBuilder tmpSb = new StringBuilder();
        for (int i = fi + 1; i < s.length(); i++) {
            if (s.charAt(i) == '-') {
                continue;
            } else {
                if (s.charAt(i) >= 'A' && s.charAt(i) <= 'Z') {
                    countDx += 1;
                } else if (s.charAt(i) >= 'a' && s.charAt(i) <= 'z') {
                    countXx += 1;
                }
            }
            tmpSb.append(s.charAt(i));
            countSum += 1;

            if (countSum == k) {
                if (countDx == countXx) {
                    sb.append(tmpSb);
                } else if (countDx > countXx) {
                    sb.append(tmpSb.toString().toUpperCase());
                } else {
                    sb.append(tmpSb.toString().toLowerCase());
                }

                if (i != s.length() - 1) {
                    sb.append('-');
                }

                countDx = 0;
                countXx = 0;
                countSum = 0;

                tmpSb = new StringBuilder();
            }
        }

        if (countSum > 0) {
            if (countDx == countXx) {
                sb.append(tmpSb);
            } else if (countDx > countXx) {
                sb.append(tmpSb.toString().toUpperCase());
            } else {
                sb.append(tmpSb.toString().toLowerCase());
            }
        }

        return sb.toString();
    }

    public static void main(String[] args) {
//        String s = "12abc-abCABc-4aB@";
//        int k = 3;

        String s = "abcs@1323-778776abZZZf-udfAd2f";
        int k = 0;

        Question1 question1 = new Question1();
        String string = question1.splitString(s, k);
        System.out.println(string);
    }
}
